Skip to content

revert: Downgrade NVRx / Upgrade nemo-run#2668

Merged
chtruong814 merged 3 commits intomainfrom
chtruong/skip-nvrx
Mar 6, 2026
Merged

revert: Downgrade NVRx / Upgrade nemo-run#2668
chtruong814 merged 3 commits intomainfrom
chtruong/skip-nvrx

Conversation

@chtruong814
Copy link
Copy Markdown
Contributor

@chtruong814 chtruong814 commented Mar 5, 2026

What does this PR do ?

revert: Downgrade NVRx / Upgrade nemo-run

The nvrx test has been failing since that got meged. Reverting for now until we can fix it better.

#2623

Changelog

  • Add specific line by line info of high level changes in this PR.

GitHub Actions CI

See the CI sectionin the Contributing doc for how to trigger the CI. A Nvidia developer will need to approve and trigger the CI for external contributors.

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

If you haven't finished some of the above items you can still open "Draft" PR.

Additional Information

  • Related to # (issue)

Summary by CodeRabbit

  • Tests
    • Added test markers to improve test organization and management.

@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai bot commented Mar 5, 2026

📝 Walkthrough

Walkthrough

A pytest marker decorator @pytest.mark.pleasefixme is added to the test_nvrx_straggler_detection_end_to_end function in the test file. The function signature and test logic remain unchanged. This annotation marks the test for future attention without modifying its behavior.

Changes

Cohort / File(s) Summary
Test Marker Addition
tests/functional_tests/training/test_nvrx_straggler.py
Added @pytest.mark.pleasefixme decorator to test_nvrx_straggler_detection_end_to_end function to flag the test for later review or fixes.

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~2 minutes

Suggested reviewers

  • yaoyu-33
  • thomasdhc
🚥 Pre-merge checks | ✅ 3 | ❌ 1

❌ Failed checks (1 warning)

Check name Status Explanation Resolution
Title check ⚠️ Warning The PR title 'revert: Downgrade NVRx / Upgrade nemo-run' does not match the actual changes, which only add a pytest marker to skip a test. The objectives indicate the intention is to skip the nvrx straggler test, not to downgrade/upgrade packages. Update the PR title to accurately reflect the change, such as 'ci: Skip nvrx straggler test until fixed' which aligns with the stated objectives.
✅ Passed checks (3 passed)
Check name Status Explanation
Docstring Coverage ✅ Passed Docstring coverage is 100.00% which is sufficient. The required threshold is 80.00%.
Test Results For Major Changes ✅ Passed PR adds only a single test skip decorator with no impact on functionality, performance, or behavior—qualifying as a minor change requiring no test documentation.
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.

✏️ Tip: You can configure your own custom pre-merge checks in the settings.

✨ Finishing Touches
  • 📝 Generate docstrings (stacked PR)
  • 📝 Generate docstrings (commit on current branch)
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch chtruong/skip-nvrx

Tip

Try Coding Plans. Let us write the prompt for your AI agent so you can ship faster (with fewer bugs).
Share your feedback on Discord.


Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands and usage tips.

Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents
Verify each finding against the current code and only fix it if needed.

Inline comments:
In `@tests/functional_tests/training/test_nvrx_straggler.py`:
- Line 241: The test file uses the decorator `@pytest.mark.pleasefixme` but never
imports pytest, causing an F821; add an import statement for pytest (e.g.,
import pytest) near the other test imports so the symbol pytest used by the
decorator is defined and the test module can be imported; update any import
grouping in the file to keep imports organized around the existing test helpers
and fixtures.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

Run ID: 55988e11-47b5-4f55-b3b2-aeb578c9b3e6

📥 Commits

Reviewing files that changed from the base of the PR and between b730ab6 and 60892f2.

📒 Files selected for processing (1)
  • tests/functional_tests/training/test_nvrx_straggler.py

return file_handler


@pytest.mark.pleasefixme
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Potential issue | 🔴 Critical

Add missing pytest import for the new decorator

Line 241 introduces @pytest.mark.pleasefixme, but pytest is not imported, which triggers F821 and breaks CI/module import.

✅ Proposed fix
 import argparse
 import logging
 import os
 import tempfile
 import time
 
+import pytest
 import torch
📝 Committable suggestion

‼️ IMPORTANT
Carefully review the code before committing. Ensure that it accurately replaces the highlighted code, contains no missing lines, and has no issues with indentation. Thoroughly test & benchmark the code to ensure it meets the requirements.

Suggested change
@pytest.mark.pleasefixme
import argparse
import logging
import os
import tempfile
import time
import pytest
import torch
🧰 Tools
🪛 Flake8 (7.3.0)

[error] 241-241: undefined name 'pytest'

(F821)

🪛 GitHub Actions: CICD NeMo

[error] 241-241: ruff: F821 Undefined name 'pytest'.


[warning] 241-241: ruff: Selection E266 has no effect because preview is not enabled.

🪛 Ruff (0.15.2)

[error] 241-241: Undefined name pytest

(F821)

🤖 Prompt for AI Agents
Verify each finding against the current code and only fix it if needed.

In `@tests/functional_tests/training/test_nvrx_straggler.py` at line 241, The test
file uses the decorator `@pytest.mark.pleasefixme` but never imports pytest,
causing an F821; add an import statement for pytest (e.g., import pytest) near
the other test imports so the symbol pytest used by the decorator is defined and
the test module can be imported; update any import grouping in the file to keep
imports organized around the existing test helpers and fixtures.

@@ -238,6 +238,7 @@ def setup_test_logging(log_file: str, rank: int):
return file_handler


@pytest.mark.pleasefixme
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

i think it was #2623

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let me try reverting that

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
This reverts commit 60892f2.

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
This reverts commit 3abd23a.

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
@chtruong814 chtruong814 requested a review from a team as a code owner March 6, 2026 01:23
@chtruong814 chtruong814 changed the title ci: Skip nvrx straggler test until fixed revert: Downgrade NVRx / Upgrade nemo-run Mar 6, 2026
@chtruong814
Copy link
Copy Markdown
Contributor Author

Merging now to unblock main branch

@chtruong814 chtruong814 merged commit d51e99b into main Mar 6, 2026
65 checks passed
@chtruong814 chtruong814 deleted the chtruong/skip-nvrx branch March 6, 2026 03:26
ko3n1g added a commit that referenced this pull request Mar 6, 2026
ko3n1g added a commit that referenced this pull request Mar 6, 2026
This reverts commit d51e99b.

Signed-off-by: oliver könig <okoenig@nvidia.com>
ko3n1g added a commit that referenced this pull request Mar 6, 2026
Signed-off-by: oliver könig <okoenig@nvidia.com>
copy-pr-bot bot pushed a commit that referenced this pull request Mar 19, 2026
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
copy-pr-bot bot pushed a commit that referenced this pull request Mar 19, 2026
Signed-off-by: oliver könig <okoenig@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants